import imgClassImg from '@/assets/img/dmp/img-class.png'
import imgObjDetectImg from '@/assets/img/dmp/img-obj-detect.png'
import textClassImg from '@/assets/img/dmp/text-class.png'
import textNamedEntityImg from '@/assets/img/dmp/text-named-entity.png'
import textSimilarityImg from '@/assets/img/dmp/text-similarity.png'
import textTripleImg from '@/assets/img/dmp/text-triple.png'

import { DmpSubTaskType, DmpTaskType, TaskTypeOptionItem } from './types/taskType'

export const toolsBtn = [
  // { name: '编辑标注', value: 'editMark', iconClass: 'icon-corsor-line' },
  { name: '移动', value: 'move', iconClass: '' },
  { name: '矩形', value: 'rect', iconClass: 'icon-Square-line' },
  { name: '圆', value: 'circle', iconClass: 'icon-circle-line' },
  { name: '多边形', value: 'polygon', iconClass: 'icon-Multilateral-line' },
  { name: '实线', value: 'line', iconClass: 'icon-line-line' },
  { name: '虚线', value: 'dottedLine', iconClass: 'icon-Dottedline-line' },
  { name: '点', value: 'point', iconClass: 'icon-point-line' }
]

export const taskTypeOptions: TaskTypeOptionItem[] = [
  {
    label: '图片',
    value: DmpTaskType.img,
    children: [
      {
        label: '图片分类',
        des: '识别图片中每个物体的位置及属性',
        imgUrl: imgClassImg,
        value: DmpSubTaskType.imgClass
      },
      {
        label: '目标检测',
        des: '根据图片中的物体划分出不同区域',
        imgUrl: imgObjDetectImg,
        value: DmpSubTaskType.imgObjDetect
      }
    ]
  },
  {
    label: '文本',
    value: DmpTaskType.text,
    children: [
      {
        label: '文本分类',
        des: '按照标签对文本内容进行分类处理',
        imgUrl: textClassImg,
        value: DmpSubTaskType.textClass
      },
      {
        label: '命名实体',
        des: '标注文本中的实体片段，如“时间”等',
        imgUrl: textNamedEntityImg,
        value: DmpSubTaskType.textNamedEntity
      },
      {
        label: '文本三元组',
        des: '标注实体片段和实体间的关系',
        imgUrl: textTripleImg,
        value: DmpSubTaskType.textTriple
      },
      {
        label: '文本相似度',
        des: '比较两段文本在语义上的相似度',
        imgUrl: textSimilarityImg,
        value: DmpSubTaskType.textSimilarity
      }
    ]
  }
]
